import os

from llama_index.llms.deepseek import DeepSeek
from llama_index.llms.openai import OpenAI


# 定义一个大模型
def deepseek_llm():
    return DeepSeek(api_key=os.environ["DEEPSEEK_API_KEY"],
                 model="deepseek-chat",
                 api_base=os.getenv("DEEPSEEK_BASE_URL"),
                 temperature=0.6)

def openAi_llm(**kwargs):
    return OpenAI(api_key=os.environ["OPENAI_API_KEY"],
                 model="gpt-4o",
                 api_base=os.getenv("OPENAI_BASE_URL"),
                 temperature=0.6,
                 **kwargs)

def qwen_llm(model,**kwargs):
    return OpenAI(api_key=os.environ["MY_QWEN_API_KEY"],
                 model=model,
                 api_base=os.getenv("MY_QWEN_BASE_URL"),
                 temperature=0.6)